package study.算法Algorithm.常用10种算法.KMP算法;

import java.util.Arrays;

/*

    kmp算法没看懂！！
    参考资料：https://www.cnblogs.com/ZuoAndFutureGirl/p/9028287.html

 */
public class KMP {
    public static void main(String[] args) {
        String str1 = "BBC ABCDAB ABCDABCDABDE";
        String str2 = "ABCDABD";

        int[] ints = kmpSet("ABCDABD");
        System.out.println("ints = " + Arrays.toString(ints));

        int index = kmpSearch(str1, str2, ints);
        System.out.println("index = " + index);
    }

    //步骤：1、得到字串的部分匹配表
    //      2、使用部分匹配表完成KMP

    /**
     * @param str1 原字符串
     * @param str2 需要匹配的字符串
     * @param next 部分匹配表
     * @return 返回第一个匹配的下标  如果没有匹配则返回-1
     */
    public static int kmpSearch(String str1, String str2, int[] next) {

        for (int i = 0, j = 0; i < str1.length(); i++) {
            //处理str1.charAt(i) ！= str2.charAt(j)时的情况   KMP核心点
            while (j > 0 && str1.charAt(i) != str2.charAt(j)){
                j = next[j - 1];
            }
            if (str1.charAt(i) == str2.charAt(j)) {
                j++;
            }
            if (j == str2.length()) {  //找到了
                return i - j + 1;
            }
        }
        return -1;
    }

    //获取一个字符串（子串）的部分匹配表   部分匹配值就是  前缀和后缀匹配的字符个数
    //例如  单词cat  这个词前缀有：c,ca  后缀有：t, at。  这个词的部分匹配值为0
    public static int[] kmpSet(String dest) {
        //创建一个next数组 保存部分匹配值
        int[] next = new int[dest.length()];
        next[0] = 0;  //如果子串的长度为1的话  那么它的部分匹配值就是0
        for (int i = 1, j = 0; i < dest.length(); i++) {

            //这是kmp算法的核心
            while (j > 0 && dest.charAt(i) != dest.charAt(j)) {
                j = next[j - 1];
            }
            //当此条件满足时，部分匹配值就+1
            if (dest.charAt(i) == dest.charAt(j)) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }
}
